package crond

import (
	"context"
	"sale-admin/config/mysql"
	"sale-admin/internal/models"

	sharding "192.168.1.75/go-pkg/db-sharding"
	"github.com/golang-module/carbon/v2"
)

// var TableType = map[string]sharding.ShardCond{
// 	"ma_ai_service_face_count": { // 曝光量
// 		Number: 2000 * 10000,
// 	},
// 	"stare_count": { // 凝视
// 		Number: 2000 * 10000,
// 	},
// 	"stare_age_gender": { // 凝视年龄性别
// 		Number: 2000 * 10000,
// 	},
// }

func ShardTable() error {
	db := mysql.NewDB()
	tableStartTime := carbon.Now().Tomorrow().StartOfDay().Timestamp() // 0点
	sharding.ShardTable(context.Background(), db, 2000*10000, models.MaAIServiceFaceCountTpl{}, tableStartTime)
	sharding.ShardTable(context.Background(), db, 2000*10000, models.StareAgeGenderTpl{}, tableStartTime)
	sharding.ShardTable(context.Background(), db, 2000*10000, models.StareCountTpl{}, tableStartTime)
	return nil

}
